FROM nvcr.io/nvidia/tensorflow:21.10-tf2-py3
RUN apt-get update
RUN pip3 install gin-config==0.1.1 absl-py==0.15.0 tensorflow_model_optimization